Fix a function change I missed

whoops