Fixes tests for ci