Fix problem with shape parameter