Add extension to test helper file (tks @srpelissier)