Small bug fix before release.