Move tests to component