Corrected associations to address run time errors