[FIX] Fix setup. Cover for some reason is not working in tests