Gradient module fix