Fix template remove bug