Fix a small bug (take into account height param)