Fix script/test for CI