fix branch handing in if __name__ ...