diff --git a/fixtracks/widgets/detectiontimeline.py b/fixtracks/widgets/detectiontimeline.py
index f202784..137d269 100644
--- a/fixtracks/widgets/detectiontimeline.py
+++ b/fixtracks/widgets/detectiontimeline.py
@@ -311,8 +311,8 @@ def main():
         df = pickle.load(f)
     data = TrackingData()
     data.setData(as_dict(df))
-    data.setUserSelection(np.arange(0,100, 1))
-    data.setAssignmentStatus(True)
+    data.setSelection(np.arange(0,100, 1))
+    data.setUserLabeledStatus(True)
     start_x = 0.1
     app = QApplication([])
     window = QWidget()