modification so that module returns properly