Fix tests

I have broken the tests. Now they work again.

Signed-off-by: Konrad Reiche <konrad.reiche@gmail.com>