Tests not working for learning switch