fix test loading issue