Fix template unit test