Fix a missing assigment