Update to work with Grunt 0.4.x