fix for issue 34

I encountered [this](https://github.com/batterseapower/test-framework/issues/34) issue while using test-framework and this change resolves it for me.