fix one-off bug