Fix single scene loading
Fix visualization script.

It was broken after a few things were deprecated.