fixed one todo