Fixing testing on CI