Small improvement to figure rendering