update code - fix