Fix test error, one more time