add module declaration to custom vault implementation