diff --git a/storage/perfschema/unittest/stub_pfs_global.h b/storage/perfschema/unittest/stub_pfs_global.h

index 8a1f9216ba2..3d7d818cc51 100644

--- a/storage/perfschema/unittest/stub_pfs_global.h

+++ b/storage/perfschema/unittest/stub_pfs_global.h

@@ -24,6 +24,9 @@

 #include <my_sys.h>

 #include <pfs_global.h>

 #include <string.h>

+#ifdef HAVE_MEMALIGN

+# include <malloc.h>

+#endif

 bool pfs_initialized= false;

@@ -43,7 +46,15 @@ void *pfs_malloc(size_t size, myf)

   if (--stub_alloc_fails_after_count <= 0)

     return NULL;

+#ifndef PFS_ALIGNEMENT

   void *ptr= malloc(size);

+#elif defined HAVE_MEMALIGN

+  void *ptr= memalign(PFS_ALIGNEMENT, size);

+#elif defined HAVE_ALIGNED_MALLOC

+  void *ptr= _aligned_malloc(size, PFS_ALIGNEMENT);

+#else

+# error "Missing implementation"

+#endif

   if (ptr != NULL)

     memset(ptr, 0, size);

   return ptr;