FIX python 2 & 3 compatibility issue