fixed a  little code