All tests passed, so add some tests back