I used the optical amplifier component (which acts as a black box) in your design by replacing the EDFA and got the aforementioned result.I set the value for the gain 50dB and the value for noise figure 4 dB and the amplification was somehow close the the amplification obtained using your EDFA. So if you intend to get 4dB NF, I think you are getting it now (maybe even less).