update to fix code