diff --git a/check_sdram_init.sh b/check_sdram_init.sh
index 43ec732..48756be 100755
--- a/check_sdram_init.sh
+++ b/check_sdram_init.sh
@@ -3,8 +3,8 @@
 GOODPHY=`grep csr_base,ddrphy csr.csv | awk -F, '{ print $3 }'`
 GOODSDRAM=`grep csr_base,sdram csr.csv | awk -F, '{ print $3 }'`
 
-USEPHY=`grep ddrphy_base *soc.py | awk '{ print $3 }'`
-USESDRAM=`grep sdram_dfii_base *soc.py | awk '{ print $3 }'`
+USEPHY=`grep ddrphy_base *soc.py | awk '{ print $3 }' | sed -e 's/,.*//'`
+USESDRAM=`grep sdram_dfii_base *soc.py | awk '{ print $3 }' | sed -e 's/,.*//'`
 
 RET=0