If you are not comfortable diving with a buddy on your own in the same places you trained the AOW class is a bad idea IMO. What specifically do you not feel ok with it? Is it Nav in general or certain areas of it? I definitely would not go back to the shop that trained you. It sounds like you did not get a good foundation. Before doing AOW I'd try to find a good underwater nav class. You have no business doing the deep dive in the AOW class with this level of discomfort.
I have recommended and taught the Nav class I offer to several people who will likely not take AOW and as a primer for those who will due to the focus on buddy skills, communication, and buoyancy and trim I put in the class. I also recommend rescue before AOW to most people; especially when they have not had any rescue skills in the OW class. (Tired diver tow is not rescue BTW IMO). For those who have had it in their OW class I have an entire dive in my AOW class to introduce rescue skills and buddy assisting.
Your feelings are all too common unfortunately. Abbreviated courses, insufficient time in confined water, and low task loading combined with short checkouts often result in the feelings you describe. Find an independent instructor that will work with you and tailor instruction to your needs. Try to stay away from dive centers when doing this unless you get a really good recommendation or when you interview the instructor you get a good feeling about their answers.
And you should interview the instructor as you would any employee and as if they were going to train your kids. I have an entire chapter on how to do that in my book as well as how to choose a shop to train with and when and what training to take.