Update junit tests