I think it might be as simple as replacing the `2` for `0` in line [556 in core.py](https://github.com/sandialabs/WecOptTool/blob/da304988106f46b8a15a963729635449b8fc63d6/wecopttool/core.py#L556). `transfer_func0 = np.expand_dims(hydrostatic_stiffness, 2)` The testing and docstrings need to be double checked too.