diff --git a/src/hwcaps.c b/src/hwcaps.c
index 6557bd4f..b5e5940a 100644
--- a/src/hwcaps.c
+++ b/src/hwcaps.c
@@ -34,7 +34,7 @@ bool CpuFeatures_IsHwCapsSet(const HardwareCapabilities hwcaps_mask,
 
 #ifdef CPU_FEATURES_TEST
 // In test mode, hwcaps_for_testing will define the following functions.
-HardwareCapabilities CpuFeatures_GetHardwareCapabilities(void);
+uintptr_t CpuFeatures_GetHardwareCapabilitiesFor(uint32_t type);
 #else
 
 // Debug facilities
@@ -148,11 +148,11 @@ uintptr_t CpuFeatures_GetHardwareCapabilitiesFor(uint32_t type) {
   return hwcaps;
 }
 
+#endif  // CPU_FEATURES_TEST
+
 HardwareCapabilities CpuFeatures_GetHardwareCapabilities(void) {
   HardwareCapabilities capabilities;
   capabilities.hwcaps = CpuFeatures_GetHardwareCapabilitiesFor(AT_HWCAP);
   capabilities.hwcaps2 = CpuFeatures_GetHardwareCapabilitiesFor(AT_HWCAP2);
   return capabilities;
 }
-
-#endif  // CPU_FEATURES_TEST