diff --git a/src/openrct2/paint/tile_element/TileElement.cpp b/src/openrct2/paint/tile_element/TileElement.cpp
index 2dab46b63a..a5bced5e34 100644
--- a/src/openrct2/paint/tile_element/TileElement.cpp
+++ b/src/openrct2/paint/tile_element/TileElement.cpp
@@ -34,6 +34,7 @@
 #include "../Supports.h"
 #include "Surface.h"
 #include "TileElement.h"
+#include "VirtualFloor.h"
 
 #ifdef __TESTPAINT__
 uint16 testPaintVerticalTunnelHeight;
diff --git a/src/openrct2/paint/tile_element/TileElement.h b/src/openrct2/paint/tile_element/TileElement.h
index a037cabfe2..3c2da871ec 100644
--- a/src/openrct2/paint/tile_element/TileElement.h
+++ b/src/openrct2/paint/tile_element/TileElement.h
@@ -109,7 +109,6 @@ void scenery_paint(paint_session * session, uint8 direction, sint32 height, rct_
 void fence_paint(paint_session * session, uint8 direction, sint32 height, rct_tile_element* tileElement);
 void large_scenery_paint(paint_session * session, uint8 direction, uint16 height, rct_tile_element *tileElement);
 void track_paint(paint_session * session, uint8 direction, sint32 height, rct_tile_element *tileElement);
-void virtual_floor_paint(paint_session * session);
 
 #ifdef __cplusplus
 }
diff --git a/src/openrct2/paint/tile_element/virtual_floor.c b/src/openrct2/paint/tile_element/VirtualFloor.cpp
similarity index 98%
rename from src/openrct2/paint/tile_element/virtual_floor.c
rename to src/openrct2/paint/tile_element/VirtualFloor.cpp
index 9a4768a862..04b378ad8c 100644
--- a/src/openrct2/paint/tile_element/virtual_floor.c
+++ b/src/openrct2/paint/tile_element/VirtualFloor.cpp
@@ -1,4 +1,4 @@
-#pragma region Copyright (c) 2014-2017 OpenRCT2 Developers
+#pragma region Copyright (c) 2014-2018 OpenRCT2 Developers
 /*****************************************************************************
  * OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
  *
@@ -14,12 +14,13 @@
  *****************************************************************************/
 #pragma endregion
 
-#include "TileElement.h"
-#include "../Paint.h"
 #include "../../Input.h"
 #include "../../interface/Viewport.h"
 #include "../../sprites.h"
 #include "../../world/Map.h"
+#include "../Paint.h"
+#include "TileElement.h"
+#include "VirtualFloor.h"
 
 static const rct_xy_element offsets[4] =
 {
diff --git a/src/openrct2/paint/tile_element/VirtualFloor.h b/src/openrct2/paint/tile_element/VirtualFloor.h
new file mode 100644
index 0000000000..c781f4af5f
--- /dev/null
+++ b/src/openrct2/paint/tile_element/VirtualFloor.h
@@ -0,0 +1,21 @@
+#pragma region Copyright (c) 2014-2018 OpenRCT2 Developers
+/*****************************************************************************
+ * OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
+ *
+ * OpenRCT2 is the work of many authors, a full list can be found in contributors.md
+ * For more information, visit https://github.com/OpenRCT2/OpenRCT2
+ *
+ * OpenRCT2 is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * A full copy of the GNU General Public License can be found in licence.txt
+ *****************************************************************************/
+#pragma endregion
+
+#ifndef _VIRTUAL_FLOOR_H
+#define _VIRTUAL_FLOOR_H
+
+void virtual_floor_paint(paint_session * session);
+#endif