Forgot to import into generated code the module the relationship class is in