Updated to fix jshint issue