Fix the generator

Since the change from JSON to YML, this generator is broken. Update the code to make it work.
It's probably clearer to just use `YAML.load_file` here.