diff --git a/SpecAugment/spec_augment_pytorch.py b/SpecAugment/spec_augment_pytorch.py
index e83fa2c..c5835d4 100644
--- a/SpecAugment/spec_augment_pytorch.py
+++ b/SpecAugment/spec_augment_pytorch.py
@@ -41,7 +41,7 @@
 import matplotlib
 matplotlib.use('TkAgg')
 import matplotlib.pyplot as plt
-from SpecAugment.sparse_image_warp_pytorch import sparse_image_warp
+from specAugment.sparse_image_warp_pytorch import sparse_image_warp
 import torch