Updating the test to pass for 4 tests.