Add 'after submission' test