diff --git a/tests/analysis/test_forecast.py b/tests/analysis/test_forecast.py
index 485fdcf..feeb613 100644
--- a/tests/analysis/test_forecast.py
+++ b/tests/analysis/test_forecast.py
@@ -148,7 +148,7 @@ def test_preprocess_sales_Success(
     assert pipe.status == STATUS_HANDLER.SUCCESS
     assert pipe.data is not None
     df = pipe.data
-    assert len(df.columns) == 5
+    assert len(df.columns) == 6
     assert any(feat not in df.columns for feat in feature_map.keys())