Fix a missing method reference