Fix python imports for catkin