Fixing another stupid bug