Fix class name editing