fixed another bug :(