Fix build so it works again.