diff --git a/proteinworkshop/datasets/pdb_dataset.py b/proteinworkshop/datasets/pdb_dataset.py
index d569624d..0daa9d95 100644
--- a/proteinworkshop/datasets/pdb_dataset.py
+++ b/proteinworkshop/datasets/pdb_dataset.py
@@ -55,9 +55,7 @@ def __init__(
         self.split_type = split_type
         self.split_sequence_similarity = split_sequence_similiarity
         self.overwrite_sequence_clusters = overwrite_sequence_clusters
-        if split_time_frames is None:
-            self.split_time_frames = split_time_frames
-        else:
+        if self.split_type == "time_cutoff":
             try:
                 self.split_time_frames = [np.datetime64(date) for date in split_time_frames]
             except: