Fix 'restart' command

Fixes #14