Changing of statements in unit test

Change in expected output to cater to change in output when refactoring
magic string