Indeed. Put it in init function now (as in scroll-to and scroll-to-top script)