Fixed compilation issue when building