Fix for #3

Fix: #3