Cleanup shelly dependency in unit tests