Add integration tests for imports