minor fix: remove test_gemm changes