You are basing this judgement on your IDE testing, which is clearly incorrect. I have specifically re-checked the Python code, and it is not broken. Perhaps you may benefit from reading this documentation on Troubleshooting-your-solution. Best of luck!

Thanks!

Take note that almost 6K coders have solved this in Python (your language). If you can, provide some evidence in the form of test examples. Otherwise, please remember "It works on my machine" is not sufficient grounds for determining the kata is flawed.

You can solve this issue with proper regular expressions