fix issue when not using custom ghe

jkiimm · Aug 5, 2013 · 3af9779 · 3af9779
1 parent 2d4653a
commit 3af9779
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gitfiti.py b/gitfiti.py
@@ -253,7 +253,7 @@ def main():
     ghe = raw_input("Enter nothing for https://github.com/ to be used: ")
     print 'Enter your github username:'
     username = raw_input(">")
-    if ghe is None:
+    if ghe is None or ghe == "":
         git_base = "https://github.com/"
         cal = get_calendar(username)
     else:
@@ -300,7 +300,7 @@ def main():
             image = IMAGES[image]
         except: 
             image = IMAGES['kitty']
-    if ghe is None:
+    if ghe is None or ghe == "":
         output = fake_it(image, get_start_date(), username, repo, offset,
                 m*match)
     else: