Improving unit test for HelloWorld example