Fix docs example for registering tasks