Correct the federated schema [models.py]

There was some errors in the federated schema