With more practice, it should be easier to stay in the dream. Try to engage with your dream environment, and don't think about waking up. Sometimes, actively trying NOT to wake up can make you think about waking up, and therefore dream that you are waking up. Does that make sense?