more tests
raise a keyerror
another fix
clean up in the test