editing class bugfix