Clean up tests imports and docs.