Add protected attributes for unit tests