Fixed the AABB issue