fix tests in the dragnet branch