fix for junit test