Update tests and grunt file.