Try to improve unit tests

where does this Class comes from?