Fix a assignment that was dropped in the last commit