Fixed edge case bug
Fixed edge case bug