ooops. Fix bug