fixed another little bug:)