Fix wdb pytest plugin