It's not until Loki points it out that Thor realizes he still has the staff in hand.

"Oh, sure," he says, and he passes it back over to Loki.

As he does so, he begins to pull a cloud above them, then tugs a piece of it off like a piece of cotton candy. It drifts towards them and settles directly above them. Thor beams up at it.