Fixes hard coded refs