Fix test: Update refs