Properly fix assignment. The old version could remove the efficiency gains provided by model eager loading.