fix the attention