fix removed class