fix incorrect class usage