Ensure script is sourced again after update