When closing to :root, properly trigger `on_return`

Fixes issue #513.

Only applies when you're using `:root`.

```ruby
close to_screen: :root
```