Fix documentation builds

Docs were calling a method on an object that wasn't defined