Updated schema since migration change