Update unit tests to reflect the change of the size getter to a function