junit test fix