modified to working test