fix #MLHR-928
fixed test case
Merge pull request #817 from gauravgopi123/MLHR-928

fix #MLHR-928