layers: In DrawState correctly return rasterSamples count for PSO
diff --git a/layers/draw_state.cpp b/layers/draw_state.cpp
index 816cf35..4476119 100644
--- a/layers/draw_state.cpp
+++ b/layers/draw_state.cpp
@@ -624,11 +624,9 @@
 static uint32_t getNumSamples(const VkPipeline pipeline)
 {
     PIPELINE_NODE* pPipe = pipelineMap[pipeline.handle];
-    // TODO: msStateCI.multisampleEnable has been removed from API, is there anything left to do here?
-//    if (VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO == pPipe->msStateCI.sType) {
-//        if (pPipe->msStateCI.multisampleEnable)
-//            return pPipe->msStateCI.rasterSamples;
-//    }
+    if (VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO == pPipe->msStateCI.sType) {
+        return pPipe->msStateCI.rasterSamples;
+    }
     return 1;
 }
 // Validate state related to the PSO