Fixed my UI fix