fixed test in yesterday's warmup problem