Fixed returning from preview not working.