diff --git a/concept_attention.py b/concept_attention.py
index 40880d3..3d13c91 100644
--- a/concept_attention.py
+++ b/concept_attention.py
@@ -527,9 +527,9 @@ class ConceptAttentionProcessor:
                                      break
                          logger.info(f"Accessed {param_count} attention-related parameters")
                          
-                     except Exception as e:
-                         logger.warning(f"Aggressive hook triggering failed: {e}")
-                         logger.info("Will use mock data as fallback")
+                    except Exception as e:
+                        logger.warning(f"Aggressive hook triggering failed: {e}")
+                        logger.info("Will use mock data as fallback")
                 else:
                     logger.warning("No suitable model found, will use mock data")