diff --git a/include/reactphysics3d/body/RigidBody.h b/include/reactphysics3d/body/RigidBody.h
index d82a019f..4c2873fc 100644
--- a/include/reactphysics3d/body/RigidBody.h
+++ b/include/reactphysics3d/body/RigidBody.h
@@ -222,6 +222,7 @@ class RigidBody : public CollisionBody {
         friend class SolveHingeJointSystem;
         friend class SolveSliderJointSystem;
         friend class Joint;
+        friend class Collider;
 };
 
 /// Compute the inverse of the inertia tensor in world coordinates.
diff --git a/src/collision/Collider.cpp b/src/collision/Collider.cpp
index 514f101b..cad778b6 100644
--- a/src/collision/Collider.cpp
+++ b/src/collision/Collider.cpp
@@ -114,7 +114,7 @@ void Collider::setLocalToBodyTransform(const Transform& transform) {
 
     RigidBody* rigidBody = static_cast<RigidBody*>(mBody);
     if (rigidBody != nullptr) {
-        mBody->mWorld.mRigidBodyComponents.setIsSleeping(mBody->getEntity(), false);
+        rigidBody->setIsSleeping(false);
     }
 
     mBody->mWorld.mCollisionDetection.updateCollider(mEntity, 0);