Fix test

it contains a case whose key is wrong

initial codes should be fixed too