fix #231 also change input sizes