Should fix [#1]

It is quick fix...