fixed another bug

fixed another bug