Fix failing tests Signed-off-by: John Harris <john@johnharris.io>
Signed-off-by: Mazz Mosley <mazz@houseofmnowster.com>