Fix association mapping

Sorry, will fix this one.