Fix issue 1, change way of passing a module