1 files changed, 2 insertions, 2 deletions
diff --git a/bin/gradient b/bin/gradient
index eb79b25..8280794 100755
--- a/bin/gradient
+++ b/bin/gradient
@@ -7,8 +7,8 @@ import struct
 import sys
 import tarfile
 import matplotlib.pyplot as plt
-from dfatool import MIMOSA
-from utils import running_mean
+from dfatool.dfatool import MIMOSA
+from dfatool.utils import running_mean
 
 voltage = float(sys.argv[1])
 shunt = float(sys.argv[2])