diff --git a/tests/libpldm_base_test.cpp b/tests/libpldm_base_test.cpp
index cb36663..7d67009 100644
--- a/tests/libpldm_base_test.cpp
+++ b/tests/libpldm_base_test.cpp
@@ -1,10 +1,10 @@
-#include <string.h>
-
 #include <array>
+#include <cstdint>
 #include <cstring>
 #include <vector>
 
 #include "libpldm/base.h"
+#include "pldm_types.h"
 
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>