Same situation as yours. Drove back without any problem. Funny thing was that I showed the officer the I797 with I94 on it, he looked around and asked: "Where is your I94?".
I had some little trouble landing though. Make sure you are prepared for the questions!