It should get better.

But your suspicion could be right. It's so much easier to become lucid, when our sleep is light, or when we are already waking up. Since that's when our mind is becoming aware as well. This happens to me a lot. I get lucid WILDing, before I have a chance to fall asleep deeper. Just keep practicing, you are doing good. Happy dreams