diff --git a/CMakeLists.txt b/CMakeLists.txt
index 795d632225a2cae2eaa5d09059be78f6c47e97ab..b7bfe024160c0d4ede6249464604e65d68a168d5 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -53,7 +53,6 @@ if (NETWORK_BANDWIDTH STREQUAL "100")
       set(NETWORK_INTERFACE 100)
    endif()
    if (NOT DEFINED DATA_WIDTH)
-      message( "abc")
       set(DATA_WIDTH 64)
    endif()
    if (NOT DEFINED CLOCK_PERIOD)
diff --git a/hls/toe/rx_engine/rx_engine.cpp b/hls/toe/rx_engine/rx_engine.cpp
index d44aab810f94a0b714daad40c27bfa71561d043e..7e54d9c5982b259a15fa80aea1363d0d15e1cb06 100755
--- a/hls/toe/rx_engine/rx_engine.cpp
+++ b/hls/toe/rx_engine/rx_engine.cpp
@@ -102,7 +102,8 @@ void drop_optional_ip_header(	stream<ap_uint<4> >&	process2dropLengthFifo,
 			process2dropLengthFifo.read(length);
 			std::cout << "(Optional) Header length: " << length << std::endl;
 
-			if (WIDTH == 64)
+			//TODO for WIDTH == 128 this only works if no IP options
+			if (WIDTH == 64 || WIDTH == 128)
 			{
 				if (length > 1)
 				{
@@ -114,9 +115,9 @@ void drop_optional_ip_header(	stream<ap_uint<4> >&	process2dropLengthFifo,
 					doh_state = SHIFT;
 				}
 			}
-			if (WIDTH == 512)
+			if (WIDTH == 256 || WIDTH == 512)
 			{
-				//TODO this is hack and works for AXI_WIDTH ==  512bit
+				//TODO this is a hack and only works if there are no IP options
 				if (length == 5)
 				{
 					process2dropFifo.read(prevWord);