updating tests for imports