diff --git a/coverage.sh b/coverage.sh
new file mode 100755
index 0000000..5867692
--- /dev/null
+++ b/coverage.sh
@@ -0,0 +1 @@
+pytest --cov=laborious --cov-report=html && xdg-open htmlcov/index.html
\ No newline at end of file
diff --git a/tests/laborious/workflows/test_minimal_retrain.py b/tests/laborious/workflows/test_minimal_retrain.py
index bcb2288..b3b03b5 100644
--- a/tests/laborious/workflows/test_minimal_retrain.py
+++ b/tests/laborious/workflows/test_minimal_retrain.py
@@ -48,6 +48,7 @@ async def test_run(workflow_mock: AsyncMock, minimal_retrain: MinimalRetrain):
                 {
                     **metadata,
                     "query": input_data["query"],
+                    'datetime_columns': input_data.get('datetime_columns', [])
                 },
                 retry_policy=ANY,
                 start_to_close_timeout=ANY
diff --git a/tests/laborious/workflows/test_predictions_batch.py b/tests/laborious/workflows/test_predictions_batch.py
index 3c8fc82..e9b9bb6 100644
--- a/tests/laborious/workflows/test_predictions_batch.py
+++ b/tests/laborious/workflows/test_predictions_batch.py
@@ -32,7 +32,8 @@ async def test_run(workflow_mock: AsyncMock, predictions_batch: PredictionsBatch
         'query': 'SELECT * FROM test',
         'schema': 'test_schema',
         'table_name': 'test_table',
-        'opc_output_config': 'test_opc_output_config'
+        'opc_output_config': 'test_opc_output_config',
+        'datetime_columns': ['timestamp', 'created_at']
     }
 
     await predictions_batch.run(input_data)
@@ -43,6 +44,7 @@ async def test_run(workflow_mock: AsyncMock, predictions_batch: PredictionsBatch
             {
                 **metadata,
                 'query': input_data['query'],
+                'datetime_columns': input_data.get('datetime_columns', [])
             },
             retry_policy=ANY,
             start_to_close_timeout=ANY