Fix live-reloading!

Closes #38