Fix minor error on test