Improve unicorn restart