diff --git a/tests/brevitas/graph/test_quantize.py b/tests/brevitas/graph/test_quantize.py
index 8f8ae3c93..3a92a3e28 100644
--- a/tests/brevitas/graph/test_quantize.py
+++ b/tests/brevitas/graph/test_quantize.py
@@ -1,4 +1,5 @@
 import copy
+import platform
 
 import pytest_cases
 import torch