Updated the test to have an assert.