If your ball falls off the tee- Rules of Golf tip

[setter02]

Applies if you haven't already taken a swing with an attempt to hit the ball. If you have, missed and address again and knock off, that's a stroke as the ball was in play. So yes, you can take a penalty stroke on the tee box.