Fixed incorrect class