Fix problem with auto-reload on source change