diff --git a/pydpu/storage.py b/pydpu/storage.py
index 51718fa..238ed59 100644
--- a/pydpu/storage.py
+++ b/pydpu/storage.py
@@ -157,6 +157,7 @@ def create(self, address):
                     nvme_controller_id=str(self.id),
                     nvme_controller=frontend_nvme_pb2.NvmeController(
                         spec=frontend_nvme_pb2.NvmeControllerSpec(
+                            trtype=opicommon_pb2.NVME_TRANSPORT_PCIE,
                             pcie_id=opicommon_pb2.PciEndpoint(
                                 physical_function=wrappers_pb2.Int32Value(
                                     value=self.pf
@@ -288,7 +289,6 @@ def update(self, address):
                     nvme_namespace=frontend_nvme_pb2.NvmeNamespace(
                         name=self.fullname,
                         spec=frontend_nvme_pb2.NvmeNamespaceSpec(
-                            trtype=opicommon_pb2.NVME_TRANSPORT_PCIE,
                             volume_name_ref="Malloc1",
                             uuid=uuid_pb2.Uuid(
                                 value="1b4e28ba-2fa1-11d2-883f-b9a761bde3fb"