diff --git a/models/demos/t3000/falcon40b/tests/test_perf_e2e_falcon.py b/models/demos/t3000/falcon40b/tests/test_perf_e2e_falcon.py
index 9630244b04dd..0aa7027a00be 100644
--- a/models/demos/t3000/falcon40b/tests/test_perf_e2e_falcon.py
+++ b/models/demos/t3000/falcon40b/tests/test_perf_e2e_falcon.py
@@ -5,6 +5,7 @@
 import torch
 import pytest
 import time
+import ttnn
 from loguru import logger
 
 from models.demos.t3000.falcon40b.reference.hf_modeling_falcon import (