Fixed compilation issue in tests