Fix to test