Add key_file element to treatment