fixed one more function