Fix silly code