Fix for #2

Fix: #2