fix so it passes tests