Small cleanup

Changed a method name