fix class remove