Update code fix