make it work with associations with custom class name