Re-factor test to only check template file name