Fix tests

Signed-off-by: Tibor Vass <teabee89@gmail.com>