Fixed changed reference in test