diff --git a/dataset/ami/ami_prepare.py b/dataset/ami/ami_prepare.py
index 44993cea..c2f88478 100644
--- a/dataset/ami/ami_prepare.py
+++ b/dataset/ami/ami_prepare.py
@@ -32,7 +32,7 @@ import json
 from ami_splits import get_AMI_split
 from distutils.util import strtobool
 
-from utils.dataio import (
+from dataio import (
     load_pkl,
     save_pkl, )