Ensure entity is connected to the right db