- fix invalid return type