small modifications to test