1 files changed, 5 insertions, 13 deletions
diff --git a/tests/tpm_lite/earlyextend.c b/tests/tpm_lite/earlyextend.c
index b46f17fc..837960c1 100644
--- a/tests/tpm_lite/earlyextend.c
+++ b/tests/tpm_lite/earlyextend.c
@@ -11,24 +11,16 @@
 #include <stdlib.h>
 
 #include "tlcl.h"
-
-#define INDEX0 0xcafe
+#include "tlcl_tests.h"
 
 int main(int argc, char** argv) {
   uint8_t value_in[20];
   uint8_t value_out[20];
-  uint32_t result;
 
   TlclLibInit();
-  TlclStartup();
-  TlclContinueSelfTest();
-
-  do {
-    result = TlclExtend(1, value_in, value_out);
-    printf("result of Extend = %d\n", result);
-  } while (result == TPM_E_DOING_SELFTEST ||
-           result == TPM_E_NEEDS_SELFTEST);
-
-  printf("Test completed successfully\n");
+  TPM_CHECK(TlclStartup());
+  TPM_CHECK(TlclContinueSelfTest());
+  TPM_CHECK(TlclExtend(1, value_in, value_out));
+  printf("TEST SUCCEEDED\n");
   exit(0);
 }