Fix tests for commit 28feb5

Signed-off-by: Martin Milata <mmilata@redhat.com>