Update to work with grunt 0.4