Fix up document package unit tests