Improve minor bug fix.