diff --git a/main.go b/main.go
index aac8901..d837d38 100755
--- a/main.go
+++ b/main.go
@@ -360,8 +360,9 @@ func draw() {
 		}
 		simpleMat.SetUnifMat4("modelMat", &tempModelMat.Translate(gglm.NewVec3(float32(rowSize), -1, 0)).Mat4)
 	}
-
 	simpleMat.SetUnifMat4("modelMat", &modelMat.Mat4)
+
+	drawUI()
 	window.SDLWin.GLSwap()
 }
 
diff --git a/meshes/mesh.go b/meshes/mesh.go
index 000a0d3..05f05be 100755
--- a/meshes/mesh.go
+++ b/meshes/mesh.go
@@ -38,7 +38,7 @@ func NewMesh(name, modelPath string, postProcessFlags asig.PostProcess) (*Mesh,
 	mesh.Buf.SetLayout(layoutToUse...)
 
 	var values []float32
-	if len(sceneMesh.ColorSets) > 0 {
+	if len(sceneMesh.ColorSets) > 0 && len(sceneMesh.ColorSets[0]) > 0 {
 		values = interleave(
 			arrToInterleave{V3s: sceneMesh.Vertices},
 			arrToInterleave{V3s: sceneMesh.Normals},