diff --git a/realm/realm_payload_main.c b/realm/realm_payload_main.c
index 131f49b..3339b9a 100644
--- a/realm/realm_payload_main.c
+++ b/realm/realm_payload_main.c
@@ -368,6 +368,12 @@
 		case REALM_SME_UNDEF_ABORT:
 			test_succeed = test_realm_sme_undef_abort();
 			break;
+		case REALM_ATTESTATION:
+			test_succeed = test_realm_attestation();
+			break;
+		case REALM_ATTESTATION_FAULT:
+			test_succeed = test_realm_attestation_fault();
+			break;
 		default:
 			realm_printf("%s() invalid cmd %u\n", __func__, cmd);
 			break;