fixed up failing test