If you lose the dream from something like that, you can always try to reenter it. Don't move stay calm, and visualize what you were just doing. Also, be sure to reality check whenever you wake from a dream suddenly. There is a good chance you are having a false awakening.