small fix for edge case behavior