diff --git a/ptsl/engine.py b/ptsl/engine.py
index b82eebd..491ad89 100644
--- a/ptsl/engine.py
+++ b/ptsl/engine.py
@@ -936,8 +936,7 @@ def select_tracks_by_name(self, names: List[str],
         op = ops.SelectTracksByName(track_names=names, selection_mode=mode)
         self.client.run(op)
 
-    def get_timeline_selection(self, format: pt.TrackOffsetOptions =
-                               pt.TrackOffsetOptions.TimeCode
+    def get_timeline_selection(self, format: TrackOffsetOptions = TimeCode
                                ) -> Tuple[str, str]:
         """
         Returns data about the current timeline selection.