Fixed load issue
Fixed break

Signed-off-by: angathorion <tanghuansong@gmail.com>
Merge pull request #34 from angathorion/load-fix

Load fix