diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
index d7daf1e5..fd0879f7 100644
--- a/nbs/src/nixtla_client.ipynb
+++ b/nbs/src/nixtla_client.ipynb
@@ -2946,11 +2946,11 @@
     "        max_wait_time=max_wait_time,\n",
     "    )\n",
     "    mock_nixtla_client._make_request = side_effect\n",
-    "    init_time = time()\n",
+    "    init_time = time.time()\n",
     "    test_fail(\n",
     "        lambda: mock_nixtla_client.forecast(df=df, h=12, time_col='timestamp', target_col='value'),\n",
     "    )\n",
-    "    total_mock_time = time() - init_time\n",
+    "    total_mock_time = time.time() - init_time\n",
     "    if should_retry:\n",
     "        approx_expected_time = min((max_retries - 1) * retry_interval, max_wait_time)\n",
     "        upper_expected_time = min(max_retries * retry_interval, max_wait_time)\n",