actually add check for return type