make fixes to the impute test