Updated with correct assignment