diff --git a/shell/.shell-common b/shell/.shell-common
index 11957d6..9c5d513 100644
--- a/shell/.shell-common
+++ b/shell/.shell-common
@@ -10,3 +10,6 @@ alias grep='grep --color=auto'
 
 # Class picker
 alias c="nli class-list"
+
+# Connect upstream back
+git branch --set-upstream-to=origin/$(git branch --show-current) $(git branch --show-current)