small code fix