Add missing mock import to be able to build the docs.