Remove reference of the old class