bug fixes / amends / polish

Signed-off-by: Jamie White <hello@jamie-white.com>