fix Test code