Move the initialization function at the load stage.