Fix template logic