Add back in missing line of code