fixed another code