Fix "Missing template" error on tests