fix one more issue