Fix code for jshint