fixed return type incompability

sorry, missed it in last commit