Get the code working again.