prepare for full multihead support
full multi-head support