Update to load the correct assembly