Updating tests

Signed-off-by: Richard Brown <chronosafe@gmail.com>