Add context usage back to test template.