little code fix