Update & fix bug