Class error message fix