Changed to correct branch function