Fix bug in gradient size