trying to fix assocations