updated edge case