Feature Restriction off for teachers

I noticed when “restricting features” for students on an assignment, when I go to grade it, I also have restricted tools. Just wondering if this is something that can be fixed? Shouldn’t the teacher always have access to all tools? I love that feature, but it would be nice to be able to give feedback with all tools and not just the ones I decided to restrict for students on a particular assignment.