Fix test reference