Update to use test