fix test, still a bug to fix