Fixed an incorrect class reference