Upgrade to work with Grunt v0.4.0