Fix returntype