Fixed a test checking for a string instead of a more reliable dict