1 files changed, 2 insertions, 5 deletions
diff --git a/tests/test_msg_flags.cpp b/tests/test_msg_flags.cpp
index 535a7d5..6c2b819 100644
--- a/tests/test_msg_flags.cpp
+++ b/tests/test_msg_flags.cpp
@@ -18,12 +18,9 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include <assert.h>
-#include <string.h>
+#include "testutil.hpp"
 
-#include "../include/xs.h"
-
-int main (int argc, char *argv [])
+int XS_TEST_MAIN ()
 {
     //  Create the infrastructure
     void *ctx = xs_init (1);