update to fix bug