diff --git a/realm/realm_payload_main.c b/realm/realm_payload_main.c
index aaf7479..e1f78c5 100644
--- a/realm/realm_payload_main.c
+++ b/realm/realm_payload_main.c
@@ -143,6 +143,12 @@
 		case REALM_SVE_UNDEF_ABORT:
 			test_succeed = test_realm_sve_undef_abort();
 			break;
+		case REALM_SME_ID_REGISTERS:
+			test_succeed = test_realm_sme_read_id_registers();
+			break;
+		case REALM_SME_UNDEF_ABORT:
+			test_succeed = test_realm_sme_undef_abort();
+			break;
 		default:
 			realm_printf("%s() invalid cmd %u\n", __func__, cmd);
 			break;