One more fix for the migrations