Remove test import