Fix test to be correct