Forgot to add the migraions; Also fixing a bug