Fixes some visualization specs