Fix small bug before release