removing uneccessary test in the class file