fixed minor bug with tests