diff --git a/tests/test_safety.py b/tests/test_safety.py
index d604bd0..d4afbe9 100644
--- a/tests/test_safety.py
+++ b/tests/test_safety.py
@@ -4,6 +4,7 @@ import os
 import struct
 import tempfile
 import unittest
+import zipfile
 from unittest import mock
 
 import remote_packages