Fixing edge case