Fixing simple compilation error