Fixed issue 14

This should fix the error :)