Fix python version check