add test in the correct file