Fix test temporarily

Fixing this so identity build #1956 can pass, will rewrite test at later date.