clean module usage, don't need to create a wrapper class indeed