fix tests and make test for divertor