updated so the tests passes -- fuck everything about doc tests