Just a minor fixes on test.