Fixed testclass to match created methods