edge case bug fixed