fix unit test

Signed-off-by: Smita Kumari <kumarismita62@gmail.com>