1 files changed, 3 insertions, 1 deletions
diff --git a/src/references.h b/src/references.h
index 310ff9f..f075bbb 100644
--- a/src/references.h
+++ b/src/references.h
@@ -1,6 +1,7 @@
 #ifndef CMARK_REFERENCES_H
 #define CMARK_REFERENCES_H
 
+#include "memory.h"
 #include "chunk.h"
 
 #ifdef __cplusplus
@@ -20,12 +21,13 @@ struct cmark_reference {
 typedef struct cmark_reference cmark_reference;
 
 struct cmark_reference_map {
+  cmark_mem *mem;
   cmark_reference *table[REFMAP_SIZE];
 };
 
 typedef struct cmark_reference_map cmark_reference_map;
 
-cmark_reference_map *cmark_reference_map_new(void);
+cmark_reference_map *cmark_reference_map_new(cmark_mem *mem);
 void cmark_reference_map_free(cmark_reference_map *map);
 cmark_reference *cmark_reference_lookup(cmark_reference_map *map,
                                         cmark_chunk *label);