minor fixed for code neat