add relevant import for mysql

fix python import

does not work without it