FIX: alphas = alphas[0] for single output