fix incorrect class reference