Fix preview mode