Fixing template test addition to use the proper hook