Let the tests be closer to the real code