Fix returning incorrect AABBs