fixing class migration bug