Fix in stage initialization