fix scalability bug

Signed-off-by: Fei Guan <feiguan@172-16-189-96.dynapool.nyu.edu>