Small tweak to fix what I broke with the last commit