Tweak preview logic to mirror behaviour elsewhere

This is closer to how things are done in DocumentsController