Fix incorrect function in previous commit