Use a cleaner way to load polymorphic parents