Small fixes in unit tests.

Signed-off-by: Lukasz Gromanowski <lgromanowski@gmail.com>