Fixed a bug with the migrations