Fix associated test