Corrected test to use the new template