Fixed minor compilation issue

Fixed minor compilation issue