Fix smaller python issues