tests passing
fix test