Get rid of push_bias_initialization