fix aabb and shader code