Fix for TF heads when TF is not present