fixed JUnits in 3 files