Fix a bug
Update code