3D Cube Puzzle

I had one of those 3D Bamboo puzzles sitting around that I finally tried to solve. I ended up breaking one of the pieces while trying to turn it, so I made a new one. With some minor trimming it fit the original puzzle. However, since it looked out of place, I just printed the rest of the pieces for a complete puzzle.

The puzzle consists of two different pieces, which I call A and B pieces. In total, there are 8 A pieces, and 4 B pieces. I printed the puzzle in Hatchbox Wood PLA, with