Fixed getSize() function