Fixed tests for loading dialog