Remove unchecked warnings from Junit tests