add more tests for custom keys