diff --git a/loc.py b/loc.py
index a40c383..e99e42a 100644
--- a/loc.py
+++ b/loc.py
@@ -2,7 +2,7 @@ import re
 import os
 import base64
 import requests
-from github import Github
+from github import Github, InputGitAuthor
 import datetime
 from string import Template
 import matplotlib.pyplot as plt