Fixes a bug with savefig not saving in utils.py