I would get used to the sensation of a dream ending, and use that time to remind yourself that "the next thing I see will probably be a dream." That will build the habit, like FM is saying, to perform reality checks (or better, just maintain awareness) after the "end" of a dream--whether it's a true awakening (and good DEILD situation) or just an FA.