Fixing another compilation error