Minor changes

- Fix assocaiation bug

I am really getting annoyed by this library. Might write my own