Cleanup. Also remove restitution hack.
Signed-off-by: Chris Sanders <zidane@outlook.com>