Fix silly bug in last commit