modify junit tests