Adding missing migration handler