Fixes a regression from previous commit that accidentally commented out a necessary function.