removed attributeerror exception from main