Fixed compilation in tests