Fix in docs: added missing use statement

The BaseMessage was not set in the docs