Tweak code, use Diana's endpoint to test.