diff --git a/processing/router.py b/processing/router.py
index 59d6716..d6df873 100644
--- a/processing/router.py
+++ b/processing/router.py
@@ -138,14 +138,16 @@ def is_remote_sensing_pretrained(dataset_doc: dict, problem: dict) -> bool:
     classification = (
         _problem.TaskKeyword.CLASSIFICATION in problem["problem"]["task_keywords"]
     )
+    regression = _problem.TaskKeyword.REGRESSION in problem["problem"]["task_keywords"]
     remote_sensing = (
         _problem.TaskKeyword.REMOTE_SENSING in problem["problem"]["task_keywords"]
     )
     return (
         "image" not in get_resource_types(dataset_doc)
-        and classification
+        and (classification or regression)
         and remote_sensing
-        and _problem.TaskKeyword.SEMISUPERVISED not in problem["problem"]["task_keywords"]
+        and _problem.TaskKeyword.SEMISUPERVISED
+        not in problem["problem"]["task_keywords"]
     )
 
 
@@ -161,7 +163,8 @@ def is_remote_sensing(dataset_doc: dict, problem: dict) -> bool:
         "image" in get_resource_types(dataset_doc)
         and (classification or regression)
         and remote_sensing
-        and _problem.TaskKeyword.SEMISUPERVISED not in problem["problem"]["task_keywords"]
+        and _problem.TaskKeyword.SEMISUPERVISED
+        not in problem["problem"]["task_keywords"]
     )
 
 
@@ -210,13 +213,23 @@ def is_text(dataset_doc: dict) -> bool:
 
 
 def is_semisupervised_tabular(problem: dict) -> bool:
-    remote_sensing = _problem.TaskKeyword.REMOTE_SENSING in problem["problem"]["task_keywords"]
-    return not remote_sensing and _problem.TaskKeyword.SEMISUPERVISED in problem["problem"]["task_keywords"]
+    remote_sensing = (
+        _problem.TaskKeyword.REMOTE_SENSING in problem["problem"]["task_keywords"]
+    )
+    return (
+        not remote_sensing
+        and _problem.TaskKeyword.SEMISUPERVISED in problem["problem"]["task_keywords"]
+    )
 
 
 def is_semisupervised_remote_sensing_pretrained(problem: dict) -> bool:
-    remote_sensing = _problem.TaskKeyword.REMOTE_SENSING in problem["problem"]["task_keywords"]
-    return remote_sensing and _problem.TaskKeyword.SEMISUPERVISED in problem["problem"]["task_keywords"]
+    remote_sensing = (
+        _problem.TaskKeyword.REMOTE_SENSING in problem["problem"]["task_keywords"]
+    )
+    return (
+        remote_sensing
+        and _problem.TaskKeyword.SEMISUPERVISED in problem["problem"]["task_keywords"]
+    )
 
 
 # --