Our aim is to study the role of gravitational stratification on the propagation of CMEs. In particular, we assess how it influences the speed and shape of CMEs and under what conditions the flux rope ejection becomes a CME or when it is quenched. We ran a set of MHD simulations that adopt an eruptive initial magnetic configuration that has already been shown to be suitable for a flux rope ejection. We varied the temperature of the backgroud corona and the intensity of the initial magnetic field to tune the gravitational stratification and the amount of ejected magnetic flux. We used an automatic technique to track the expansion and the propagation of the magnetic flux rope in the MHD simulations. From the analysis of the parameter space, we evaluate the role of gravitational stratification on the CME speed and expansion. Our study shows that gravitational stratification plays a significant role in determining whether the flux rope ejection will turn into a full CME or whether the magnetic flux rope will stop in the corona. The CME speed is affected by the background corona where it travels faster when the corona is colder and when the initial magnetic field is more intense. The fastest CME we reproduce in our parameter space travels at 850 km/s. Moreover, the background gravitational stratification plays a role in the side expansion of the CME, and we find that when the background temperature is higher, the resulting shape of the CME is flattened more. Our study shows that although the initiation mechanisms of the CME are purely magnetic, the background coronal plasma plays a key role in the CME propagation, and full MHD models should be applied when one focusses especially on the production of a CME from a flux rope ejection.