Forgot to turn the output back to a dictionary. No it works