Fixed a missing line of code