Fix some module references.