Modification to test