Minor fixes

Please fix this and you are sure to see that bugs in your mod. Thanks! :D