Fix failing tests due to previous commit