diff --git a/unit/test-cmac-aes.c b/unit/test-cmac-aes.c
index 70030cc9..92342a53 100644
--- a/unit/test-cmac-aes.c
+++ b/unit/test-cmac-aes.c
@@ -29,7 +29,7 @@
 #include <assert.h>
 #include <ell/ell.h>
 
-#include "src/aes.h"
+#include "src/crypto.h"
 
 struct cmac_data {
 	const void *key;
diff --git a/unit/test-eapol.c b/unit/test-eapol.c
index c9825633..8d8138a7 100644
--- a/unit/test-eapol.c
+++ b/unit/test-eapol.c
@@ -32,7 +32,6 @@
 #include "src/sha1.h"
 #include "src/eapol.h"
 #include "src/crypto.h"
-#include "src/aes.h"
 
 struct eapol_key_data {
 	const unsigned char *frame;