Update Test framework. Minor fix.
Merge pull request #9 from xmartlabs/fixes

Update Test framework. Minor fix.