Fixed error, fixed #7 i hope