Fix silly regression bug