Fix size limit warning and add unit test