diff --git a/test_massive_dqn.py b/test_massive_dqn.py
index 3d03c69..16b5e3a 100644
--- a/test_massive_dqn.py
+++ b/test_massive_dqn.py
@@ -46,12 +46,13 @@ def test_dqn_architecture():
         output = network(test_input)
         
     if isinstance(output, tuple):
-        q_values, regime_pred, price_pred, volatility_pred, features = output
+        q_values, regime_pred, price_pred, volatility_pred, features, multi_timeframe_pred = output
         print(f"   ✅ Q-values shape: {q_values.shape}")
         print(f"   ✅ Regime prediction shape: {regime_pred.shape}")  
         print(f"   ✅ Price prediction shape: {price_pred.shape}")
         print(f"   ✅ Volatility prediction shape: {volatility_pred.shape}")
         print(f"   ✅ Features shape: {features.shape}")
+        print(f"   ✅ Multi-timeframe predictions shape: {multi_timeframe_pred.shape}")
     else:
         print(f"   ✅ Output shape: {output.shape}")