fix test
code tweak