diff --git a/CMakeLists.txt b/CMakeLists.txt
index c115505..ade9108 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -60,11 +60,12 @@ if(GST_BUILD)
         add_definitions(-DRPI)
         include(${CMAKE_CURRENT_SOURCE_DIR}/cmake_modules/functions.cmake)
         findRpiRevision( RPI_REVISION )
-        set(RPI_4_REVISIONS a03111 b03111 b03112 c03111 c03112 d03114)
-        if(RPI_REVISION IN_LIST RPI_4_REVISIONS)
+	math(EXPR RPI_MODEL "(0x${RPI_REVISION}>>4)&0xFF")
+        message( "-- Raspberry Pi Model: ${RPI_MODEL}" )
+        if(RPI_MODEL EQUAL 17)
             message("Raspberry Pi 4 Found")
             add_definitions(-DPI4)
-        endif(RPI_REVISION IN_LIST RPI_4_REVISIONS)
+        endif(RPI_REVISION EQUAL 17)
     endif(RPI_BUILD)
 message(STATUS "${GST_LIBRARIES}")
 endif(GST_BUILD)