Help: Normalized Velocity Stops RigidBody's Gravity

I've been using the code below to normalize my object's magnitude. It works except for the fact that when the object is moving, the rigidbody fails to fall due to gravity.

My question is how can I modify this code so that it doesn't affect the y-component of the object's velocity?

function Update ()
{
if (rigidbody.velocity.magnitude > speed)
{
rigidbody.velocity = rigidbody.velocity.normalized * speed;
}
}

 

You should treat the gravity component separately. If it's just the Y component then that's pretty easy - save the Y component before you clamp the velocity and put it back afterwards.

 

Wow that was a remarkably easy fix. Thanks!