Fix silly issue found in testing