Fixed up 1 test