fix wrong return type