Fixed gradient bug!