Fixing error in early remove return