fixed another bug with figures