Small fixes (beta)