BLD: Fix Python 3 check