fixing junit test