I stopped hanging out with a friend who does major sins and doesn’t pray but keeps lecturing me about Islam. Was I wrong?

Before I say more, just know I tried to ask this on a proper Islamic Q&A site but the queue was full, so I’m cautious about asking here. I really only want answers based on Quran and Sunnah, not just personal opinions. I have a friend who calls himself a strong believer, but he struggles a lot with self-discipline-he doesn’t pray and he smokes marijuana. I’ve advised him so many times, had serious talks, and pointed out how wrong these things are. I still used to sit with him once a week, hoping my presence would remind him and maybe help him change. But lately, all he wants to do is discuss fiqh and religious rules with me. He studied some Islamic knowledge in the past, then quit and went off track. Sometimes he gets things wrong, and it bothers me when he tries to teach me Islam while he isn’t practicing what he knows. It feels like he’s just feeding his ego, trying to seem important and knowledgeable in front of me. So one day, when he started again, I calmly said, “Okay, it’s time for me to go,” and I left. He got upset because he probably realized why I was leaving. Was I wrong to cut him off like that? I’ve told him many times to practice what he preaches, but it only made me more frustrated seeing no change while he keeps lecturing me. To anyone who can help with this, jazakum Allahu khairan. Another thing that worries me is that I’m not a scholar. I don’t always know if what he says is correct or not, so I don’t know how to respond. It feels unsafe to learn my deen from someone whose practice doesn’t match his words.