Small update to test