Fixed test for David

All tests pass.