Add in secret and update the tests