Fixed issue with function signature conflict in tests