correct some type leading to missing module error