Fix failing XS test