diff --git a/clang/test/Sema/aarch64-sme-intrinsics/acle_sme_zero.c b/clang/test/Sema/aarch64-sme-intrinsics/acle_sme_zero.c
index 8ea80bc6568fec..a852ffa09c60e5 100644
--- a/clang/test/Sema/aarch64-sme-intrinsics/acle_sme_zero.c
+++ b/clang/test/Sema/aarch64-sme-intrinsics/acle_sme_zero.c
@@ -1,6 +1,8 @@
 // REQUIRES: aarch64-registered-target
 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sme -target-feature +sve -fsyntax-only -verify %s
 
+#include <arm_sme.h>
+
 void test_svzero_args(uint64_t m) {
   svzero_za(0); // expected-error {{too many arguments to function call, expected 0, have 1}}
   svzero_za(m); // expected-error {{too many arguments to function call, expected 0, have 1}}