modified junit test