fixing model to match migration