Fixed a test calling an incorrect function