Fix test up