gradient bug fix