Correct assignment error

Corrected error