Change class name to pass py.test