Small correction to test