Fix for #1

It is quick fix...