Fixed compilation issue

Fixed compilation issue