Fixing return of size function