Fix unit test

Change-Id: I66e0b99b1d15695ae288f4ad7d56da4e24e69b18