fixing first tests