Fixed a small bug due to a change in method names