__main__ restored