fix two edge cases