Fixed another gradient issue