added small fix to avoid test failure