fixed another little bug