Fixed a ref from test to main