Fix local tests