[Output] Fix saving figure