Fix imports in a test file