a little code fix