Fix small bug in develop