Fixing minor compilation error