py: Bug fix