Update unit test to use ->getData