Fixing couple of method calls that got missed in renaming.