Fix for #3

Hello :)