code fix bug