Fixing an oversight in the last commit