fix module weight setting