Change test to not use assertSelectCount on attribute

Have to rely on the structure, until PHPUnit works