Use parametrize for tests