Edge case bugfix