Fix the docs build.

#36 broke the docs build. This addresses that.

@mdickinson