Fixes compile error in test