Quick fix for CI