It's a nit pick but can you use the `if is_only_keys:` style like the rest of the file?