update, fix code