Fix incorrect assignment in last commit