Fixed a glaring oversight in the code.