Fix wrong return type

TBR