fix regex for edge cases