Grunt 0.4 compatibility
Fix unit tests for grunt 4.0