Type Error with Custom Layer Implementation - Effect Community