Fix minor bug in relese branch