updated to correct imports and changed test dictionary key to proper template key