fix small problem
few minor fixes