Fixing issue with preview mode & customizer.