Fixed level restarts

Now works after dying