fixed for new python syntax