Converted 'assert' to 'should' test
Added assertion