diff --git a/alfred/tests/test_analysis.py b/alfred/tests/test_analysis.py
index 62e558a..66bdf34 100644
--- a/alfred/tests/test_analysis.py
+++ b/alfred/tests/test_analysis.py
@@ -9,6 +9,7 @@
 import shutil
 import numpy as np
 import os
+import sys
 from io import StringIO
 from os import listdir
 from sklearn.decomposition import PCA