diff --git a/scripts/tinytinypy b/scripts/tinytinypy
index 70eda39..e3e99ec 100644
--- a/scripts/tinytinypy
+++ b/scripts/tinytinypy
@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
 
 import argparse
-import atexit
 from getpass import getpass
 import os
 import re
diff --git a/tinytinypy/main.py b/tinytinypy/main.py
index adf5dca..899126d 100755
--- a/tinytinypy/main.py
+++ b/tinytinypy/main.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 
+import atexit
 import http.client
 import json