Updated expected result of test to assert everything is working