Improve test by deleting useless code