Scripts load before elementor render widgets in editor mode?

[KneePham]

I'm writing a custom widget and my script seems to be loading before Elementor renders.

My main concern is I can't seem to query elements while in editor mode.

Example
PHP - widget class

protected function render() {
 ?>
  <div class="map"></div>
 <?php
}

JS

document.querySelector('.map')

On the frontend my script is able to find the element but in the editor mode, it returns null. Is there something I'm particularly missing? I've attempt to load my js file using different hooks https://code.elementor.com/php-hooks/. but that doesn't seem to help. I'm a bit at lost.