Fix test part 2