Fix minor compilation issue