Fix lazy association load