diff --git a/src/schnetpack_gschnet/datasets/qm9.py b/src/schnetpack_gschnet/datasets/qm9.py
index 3480314..3d2b68c 100644
--- a/src/schnetpack_gschnet/datasets/qm9.py
+++ b/src/schnetpack_gschnet/datasets/qm9.py
@@ -285,9 +285,9 @@ def _download_data(
 
         property_list = []
 
-        irange = np.arange(len(ordered_files), dtype=np.int)
+        irange = np.arange(len(ordered_files), dtype=int)
         if uncharacterized is not None:
-            irange = np.setdiff1d(irange, np.array(uncharacterized, dtype=np.int) - 1)
+            irange = np.setdiff1d(irange, np.array(uncharacterized, dtype=int) - 1)
 
         for i in tqdm(irange):
             xyzfile = os.path.join(raw_path, ordered_files[i])