diff --git a/firmware/stub/vboot_api_stub.c b/firmware/stub/vboot_api_stub.c
index 17f4207..e7ee136 100644
--- a/firmware/stub/vboot_api_stub.c
+++ b/firmware/stub/vboot_api_stub.c
@@ -13,6 +13,7 @@
 #include <string.h>
 #include <sys/time.h>
 
+#include "2api.h"
 #include "2common.h"
 #include "vboot_api.h"
 #include "vboot_test.h"
@@ -50,7 +51,7 @@
 
 vb2_error_t VbExGetAltFwIdxMask(void)
 {
-	return 0;
+	return VB2_SUCCESS;
 }
 
 uint32_t VbExKeyboardRead(void)
@@ -147,10 +148,10 @@
 
 vb2_error_t VbExLegacy(enum VbAltFwIndex_t altfw_num)
 {
-	return 1;
+	return VB2_SUCCESS;
 }
 
 vb2_error_t VbExSetVendorData(const char *vendor_data_value)
 {
-	return 0;
+	return VB2_SUCCESS;
 }