Fix dialog loading