diff --git a/include/container.h b/include/container.h
index a54e016a..dd934be6 100644
--- a/include/container.h
+++ b/include/container.h
@@ -68,4 +68,7 @@ swayc_t *destroy_view(swayc_t *view);
 swayc_t *find_container(swayc_t *container, bool (*test)(swayc_t *view, void *data), void *data);
 void container_map(swayc_t *, void (*f)(swayc_t *, void *), void *);
 
+//Mappings
+void set_view_visibility(swayc_t *view, void *data);
+
 #endif
diff --git a/include/layout.h b/include/layout.h
index 38a1f24b..a7f43fda 100644
--- a/include/layout.h
+++ b/include/layout.h
@@ -22,6 +22,7 @@ void arrange_windows(swayc_t *container, int width, int height);
 void unfocus_all(swayc_t *container);
 void focus_view(swayc_t *view);
 void focus_view_for(swayc_t *ancestor, swayc_t *container);
+
 swayc_t *get_focused_container(swayc_t *parent);
 swayc_t *get_swayc_for_handle(wlc_handle handle, swayc_t *parent);
 
diff --git a/sway/container.c b/sway/container.c
index 89958a4f..3cf9e47a 100644
--- a/sway/container.c
+++ b/sway/container.c
@@ -227,3 +227,15 @@ void container_map(swayc_t *container, void (*f)(swayc_t *view, void *data), voi
 	}
 }
 
+void set_view_visibility(swayc_t *view, void *data) {
+	uint32_t *p = data;
+	if (view->type == C_VIEW) {
+		wlc_view_set_mask(view->handle, *p);
+		if (*p == 2) {
+			wlc_view_bring_to_front(view->handle);
+		} else {
+			wlc_view_send_to_back(view->handle);
+		}
+	}
+	view->visible = (*p == 2);
+}
diff --git a/sway/layout.c b/sway/layout.c
index 963bd8dc..4407742a 100644
--- a/sway/layout.c
+++ b/sway/layout.c
@@ -272,4 +272,3 @@ void focus_view_for(swayc_t *top, swayc_t *view) {
 	}
 }
 
-
diff --git a/sway/workspace.c b/sway/workspace.c
index 15828658..9bc3215f 100644
--- a/sway/workspace.c
+++ b/sway/workspace.c
@@ -80,19 +80,6 @@ bool workspace_by_name(swayc_t *view, void *data) {
 		   (strcasecmp(view->name, (char *) data) == 0);
 }
 
-void set_mask(swayc_t *view, void *data) {
-	uint32_t *p = data;
-	if (view->type == C_VIEW) {
-		wlc_view_set_mask(view->handle, *p);
-	}
-	if (*p == 2) {
-		wlc_view_bring_to_front(view->handle);
-	} else {
-		wlc_view_send_to_back(view->handle);
-	}
-	view->visible = (*p == 2);
-}
-
 swayc_t *workspace_find_by_name(const char* name) {
 	return find_container(&root_container, workspace_by_name, (void *) name);
 }
@@ -198,9 +185,9 @@ void workspace_switch(swayc_t *workspace) {
 
 		// set all c_views in the old workspace to the invisible mask if the workspace
 		// is in the same output & c_views in the new workspace to the visible mask
-		container_map(focused_workspace, set_mask, &mask);
+		container_map(focused_workspace, set_view_visibility, &mask);
 		mask = 2;
-		container_map(workspace, set_mask, &mask);
+		container_map(workspace, set_view_visibility, &mask);
 		wlc_output_set_mask(ws_output->handle, 2);
 
 		destroy_workspace(focused_workspace);