You can probably remove the temporary assignment and just access `document.display_type` directly, removing it from https://github.com/alphagov/whitehall/pull/156/files#L1R42.