Fix testing wrong struct