diff --git a/contrib/checkbox-ce-oem/checkbox-provider-ce-oem/tests/test_rpmsg_load_firmware.py b/contrib/checkbox-ce-oem/checkbox-provider-ce-oem/tests/test_rpmsg_load_firmware.py
index 2704457e7f..846439926e 100755
--- a/contrib/checkbox-ce-oem/checkbox-provider-ce-oem/tests/test_rpmsg_load_firmware.py
+++ b/contrib/checkbox-ce-oem/checkbox-provider-ce-oem/tests/test_rpmsg_load_firmware.py
@@ -91,11 +91,17 @@ def test_get_rpmsg_state(self, mock_read):
         self.assertEqual(self._rpmsg_load_fw_test.rpmsg_state, expected_result)
 
     @patch("pathlib.Path.write_text")
-    def test_set_rpmsg_state(self, mock_write):
-        expected_result = "test-response"
+    def test_set_rpmsg_state_success(self, mock_write):
+        expected_result = "start"
         self._rpmsg_load_fw_test.rpmsg_state = expected_result
         mock_write.assert_called_once_with(expected_result)
 
+    def test_set_rpmsg_state_invalid_value(self):
+        with self.assertRaisesRegex(
+            ValueError, "Unsupported value for remote processor state"
+        ):
+            self._rpmsg_load_fw_test.rpmsg_state = "invalid value"
+
     def test_get_search_pattern(self):
         self.assertDictEqual(
             self._rpmsg_load_fw_test.search_pattern,