diff --git a/openchem/data/graph_data_layer.py b/openchem/data/graph_data_layer.py
index f1bbb33..07b85a2 100644
--- a/openchem/data/graph_data_layer.py
+++ b/openchem/data/graph_data_layer.py
@@ -76,7 +76,7 @@ def __init__(self,
                                                                           return_max_len=True)
             self.max_len = max_len
             if target is not None:
-                target = np.asarray(target, dtype=np.float).T
+                target = np.asarray(target, dtype=float).T
             clean_smiles = [clean_smiles[i] for i in clean_idx]
             num_atoms = [num_atoms[i] for i in clean_idx]
             self.clean_idx = clean_idx