diff --git a/omics/cli/run_analyzer/__init__.py b/omics/cli/run_analyzer/__init__.py
index a7dbe06..161763e 100644
--- a/omics/cli/run_analyzer/__init__.py
+++ b/omics/cli/run_analyzer/__init__.py
@@ -1 +1 @@
-from timeline import plot_timeline
\ No newline at end of file
+from . import timeline
diff --git a/omics/cli/run_analyzer/__main__.py b/omics/cli/run_analyzer/__main__.py
index 175c9db..591e0d5 100755
--- a/omics/cli/run_analyzer/__main__.py
+++ b/omics/cli/run_analyzer/__main__.py
@@ -46,8 +46,8 @@
 import dateutil
 import dateutil.utils
 import docopt
+import timeline  # type: ignore
 from bokeh.plotting import output_file
-from timeline import plot_timeline # type: ignore
 
 exename = os.path.basename(sys.argv[0])
 OMICS_LOG_GROUP = "/aws/omics/WorkflowLog"
@@ -528,4 +528,4 @@ def tocsv(val):
         )
         title = f"arn: {run['arn']}, name: {run.get('name')}"
 
-        plot_timeline(resources, title=title, max_duration_hrs=run_duration_hrs)
+        timeline.plot_timeline(resources, title=title, max_duration_hrs=run_duration_hrs)