Fix displacement code.