Forgot import in test class