Edited TestBase.java file