Fixing async tests