Fix memory issue in key grab API