Made trie tests work, but is an ugly hack