Fixed compile errors in tests