Fixing an obvious bug