Fixed the `setShim` method