diff --git a/tests/check_code.sh b/tests/check_code.sh
index b2b42a6..0b0a363 100755
--- a/tests/check_code.sh
+++ b/tests/check_code.sh
@@ -7,7 +7,7 @@ setup_git() {
 
 check_format() {
   black hyperglass
-  git checkout origin/master
+  git checkout master
   git add hyperglass/ *.py
   git commit --message "Black Formatting - travis #$TRAVIS_BUILD_NUMBER"
   echo "Completed Black Formatting"
@@ -16,7 +16,7 @@ check_format() {
 check_pylint() {
   PYLINT_SCORE=$(python3 manage.py pylint-badge --integer-only True)
   echo "Pylint score: $PYLINT_SCORE"
-  git checkout origin/master
+  git checkout master
   git add pylint.svg
   git commit --message "Pylint Badge - travis #$TRAVIS_BUILD_NUMBER"
   echo "Completed Pylint Check & Badge Creation"