Fixed bug in improvment.
Fix bug in improvement.