fixed JUnits in 2 test classes