Set batch size to 10 and fix strange NoneType error.