fix test controller issue