Fix the test to check for primitives