Update Junit tests