Fixed bug in improvment.