A little more cleanup

Stillpoint-Software · Sep 5, 2024 · f978488 · f978488
1 parent 80fb76a
commit f978488
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Hyperbee.AsyncExpressions/StateMachineBuilder.cs b/src/Hyperbee.AsyncExpressions/StateMachineBuilder.cs
@@ -250,9 +250,9 @@ private static bool TryGetAwaiterType( Expression expr, out Type awaiterType )
 
         static Type GetAwaiterType( Type taskType )
         {
-            var genericArgument = taskType.IsGenericType ? taskType.GetGenericArguments()[0] : typeof(void);
+            var genericArgument = taskType.IsGenericType ? taskType.GetGenericArguments()[0] : null;
 
-            if ( genericArgument == typeof(void) || genericArgument.FullName == "System.Threading.Tasks.VoidTaskResult" )
+            if ( genericArgument == null || genericArgument.FullName == "System.Threading.Tasks.VoidTaskResult" )
             {
                 return typeof(ConfiguredTaskAwaitable.ConfiguredTaskAwaiter);
             }