fixed one bug add another