minor fix to figure