can probably use assoc-in for test function