changed getter and test to make it works