Fixing unit test.

Change-Id: Ic87901325f164539ac363d76501762e41a07e94a
Signed-off-by: David DV <davidoftheold@gmail.com>