Added a second comparison for output and expectation