fix bug key return