Minor cleanup correction