fixed one more issue