Fixed the failing test

Change-Id: Ifd0bcec629503f257b094b9e6c869c97b67db2a4