diff --git a/prosit_RT_ori.py b/prosit_RT_ori.py
index b07b71d29e5b834d0ddb6b5f10065c05ca9fc254..f4544cb38b05b64fb6bc6efbe9b7d3db3bed7d2e 100644
--- a/prosit_RT_ori.py
+++ b/prosit_RT_ori.py
@@ -2,7 +2,7 @@ import os
 import wandb as wdb
 import numpy as np
 import pandas as pd
-import dlomix
+
 from dlomix import constants, data, eval, layers, models, pipelines, reports, utils
 from dlomix.models import RetentionTimePredictor
 from dlomix.data import RetentionTimeDataset