Minor simplifcation on script to convert JUnit assertions to AssertJ assertions