Fixing timeout for testing