NXDRIVE-153: Fix the UnitTest