Correct savefig import