Fix to use grunt for test