updates to fix preview