clean code in TestClass