Dirty fix for #4

That's littlebit dirty fix fro #4