Fix class name...derp