Fix some refs