Small fix for incorrect assignment.