Fixes problem with the figures