diff --git a/src/renderer/components/Experiment/Train/JobProgress.tsx b/src/renderer/components/Experiment/Train/JobProgress.tsx
index e30a04095dc0c8c6ef1544187200028bdd98f4e3..b2276da682d26ccd88f1e239c8296f34910e3b64 100644
--- a/src/renderer/components/Experiment/Train/JobProgress.tsx
+++ b/src/renderer/components/Experiment/Train/JobProgress.tsx
@@ -6,6 +6,8 @@ import { jobChipColor } from 'renderer/lib/utils';
 dayjs.extend(relativeTime);
 var duration = require('dayjs/plugin/duration');
 dayjs.extend(duration);
+import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
+
 
 export default function JobProgress({ job }) {
   return (