Fix test case

Change-Id: I4e3d311f3b7501cda93cb260ff997a3501f4c6cc