Fix test 6 problem