diff --git a/tests/tools/test_calflops.py b/tests/tools/test_calflops.py
index 17e1635..d5c92e2 100644
--- a/tests/tools/test_calflops.py
+++ b/tests/tools/test_calflops.py
@@ -5,8 +5,7 @@
 
 
 def test_calcualte_flops():
-    timm_create = BACKBONES.build({'name': 'timm_resnet50'})
-    model = timm_create(model_name='resnet50')
+    model = BACKBONES.build({'name': 'timm_resnet50'})
     flops, macs, params = calculate_flops(model, (1, 3, 224, 224))
     assert flops == '8.21 GFLOPS'
     assert macs == '4.09 GMACs'