fix for edge cases