Fix for init_bias_target_marginals bug