Of course, I am also distracted by colleagues when I am at office. It is also annoying. They do not care whether I am in focus mode or not. So, going back to the focus mode takes long and some times I couldn't until the end of the day.
Still, by answering their questions, anyhow I am unblocking them. When I WFH, I rarely get questions. Colleagues tend to send an email or slack and waiting my responses. So, my productivity may fall but overall productivity increases.
I do, however, hope companies move back to dedicated room culture. With that, at least I won't be distracted by others' hall way chats but questions.