Minor API change fix.