Fix a hackity hack

This migration was a hack job, and the models declaration was not
accurate so this commit updates that to prevent strange behaviors