Fix styling for preview mode