Update code.js

Bug fixed