Fix typing annotations in `tests.replication`.