fix class and gem which break migration