Thanks Damian for your response.
I’m using “optical amplifier” Component in Optical Amplifiers Library.
My design is quite complex it contains 3 PM-QPSK Tx and Rx and Mode selectors and Matlab comonent for Few mode fiber. However I can send it for you in private message if it can help.