Fixes tests locally