Fixed a missing assignment