Replace snapshot with state object