fixing incorrect class