amended tests as per comment