further fix association bug