Modify jUnit test