Fix references in test