Fixed unit test

Change-Id: I6cbec124c2e882d50211757ccac2dba165cad94c
Signed-off-by: Marius <marius.oe@gmail.com>