fix Test bug