Fix edge case error