Use more appropriate action hook
Merge pull request #1 from kucrut/use-load-widgets-hook

Use more appropriate action hooks, props @kucrut