Updated the tests for the result data structure