Fix template type error