Update tests for key generation method change

Fixing breaking test.