diff --git a/camkes/templates/component.simple.c b/camkes/templates/component.simple.c
index ba74a9ca..a078e662 100644
--- a/camkes/templates/component.simple.c
+++ b/camkes/templates/component.simple.c
@@ -352,7 +352,7 @@ static seL4_Error simple_camkes_get_iospace(void *data, uint16_t domainID, uint1
 }
 #endif
 
-#ifdef CONFIG_ARM_SMMU
+#ifdef CONFIG_TK1_SMMU
 static seL4_Error simple_camkes_get_iospace_cap_count(void *data, int *count) {
     if (count) {
         *count = /*? len(iospaces) ?*/;