Yeah, take the battery out, then clear the box of the vapor. Think about the battery like a lighter when you're smoking a bowl. You don't keep lighting the bowl after your hit, ya know?
When you leave the battery in, it continues the cook the bud, losing vapor, wasting the juice in the battery...