[BUG] Python 3 compabiltiy