Unittest now works with correct class