Fix test to be more accurate