instead of ``Module``, I think you should use ``cls`` here.