Remove reference to Key module