When passing another vehicle, you should return to your original lane when:

When passing another vehicle, move back into your original lane only when you can see the passed vehicle’s headlights in your rearview mirror. This ensures that you will have enough room to safely pull back in front of the other vehicle.
DMV Writen Test