Use a module instead of a wrapper