fixed imputing issue