Fix documentation unit tests for flow typing