fix callback type so that module is valid as well, since the code allows module in place of callback already