Update initOnLoad behaviour