Fixing issue with grunt task