[NGRINDER-594] Fix bugs