Fix: command prompt not ideal after a restore

Fixes #36