diff --git a/framework/python/src/api/api.py b/framework/python/src/api/api.py
index 044a72178..e978f4cf7 100644
--- a/framework/python/src/api/api.py
+++ b/framework/python/src/api/api.py
@@ -28,6 +28,7 @@
 DEVICE_MAC_ADDR_KEY = "mac_addr"
 DEVICE_MANUFACTURER_KEY = "manufacturer"
 DEVICE_MODEL_KEY = "model"
+DEVICE_TEST_MODULES_KEY = "test_modules"
 
 class Api:
   """Provide REST endpoints to manage Test Run"""
@@ -197,6 +198,7 @@ async def save_device(self, request: Request, response: Response):
         device.manufacturer = device_json.get(DEVICE_MANUFACTURER_KEY)
         device.model = device_json.get(DEVICE_MODEL_KEY)
         device.device_folder = device.manufacturer + " " + device.model
+        device.test_modules = device_json.get(DEVICE_TEST_MODULES_KEY)
 
         self._test_run.create_device(device)
         response.status_code = status.HTTP_201_CREATED