I had trouble with myself (graduated 2007, had straight A's in my major) until I realized that I do always get the work done? If you get the work done, let yourself play; you'll get uncomfortable enough later that you will buckle down?
I also make sure whatever I am doing is being done or is about a subject/perspective that I choose and really enjoy and want to learn about? Everything is connected to everything else so it can be easy and quite fun/challenging to pick something you'd like. I wrote a term paper on how the French were not inferior to the English during the Industrial Revolution :-) My professor's specialty was the French so I set about to "prove" the opposite of what one is generally taught in history. It was a lot of fun and I impressed him (though he didn't agree with all I had to say :-) I generally find my professors' thesis papers, one can find them for sale online inexpensively easily, and then I quote from them in my papers and dispute whatever I quote as if it's wrong

Bratty but they get a good chuckle and I get an A for effort/imagination.