Fixing visual updates

- I dun goofed