Fixing import in tests