Fixed Bug in TestClass