diff --git a/hipamd/src/hip_hmm.cpp b/hipamd/src/hip_hmm.cpp
index 5f636d6b39..64f2ca0435 100644
--- a/hipamd/src/hip_hmm.cpp
+++ b/hipamd/src/hip_hmm.cpp
@@ -65,6 +65,8 @@ static_assert(static_cast<uint32_t>(hipMemRangeAttributeLastPrefetchLocation) ==
 hipError_t hipMallocManaged(void** dev_ptr, size_t size, unsigned int flags) {
   HIP_INIT_API(hipMallocManaged, dev_ptr, size, flags);
 
+  CHECK_STREAM_CAPTURE_SUPPORTED();
+
   if ((dev_ptr == nullptr) || (size == 0) ||
       ((flags != hipMemAttachGlobal) && (flags != hipMemAttachHost))) {
     HIP_RETURN(hipErrorInvalidValue);