[BUGFIX] Correct usage of single branch argument