diff --git a/vega_sim/service.py b/vega_sim/service.py
index 16031f5f9..8b19b1aa0 100644
--- a/vega_sim/service.py
+++ b/vega_sim/service.py
@@ -2974,8 +2974,8 @@ def list_team_referee_history(
 
     def build_iceberg_opts(
         self, market_id: str, peak_size: float, minimum_visible_size: float
-    ) -> vega_protos.vega.IcebergOpts:
-        return vega_protos.commands.v1.IcebergOpts(
+    ) -> vega_protos.commands.v1.commands.IcebergOpts:
+        return vega_protos.commands.v1.commands.IcebergOpts(
             peak_size=num_to_padded_int(peak_size, self.market_pos_decimals[market_id]),
             minimum_visible_size=num_to_padded_int(
                 minimum_visible_size, self.market_pos_decimals[market_id]