Fix test, add an init