Fixed gradient bug.