ooops. fixing bug