Improved unit test

Change-Id: Ifda06e42a89bbfa3d5ab8caa12fbc674badae686
Signed-off-by: Matthias Steiner <matthias.steiner@inscope.net>