small fix
ref #5. Works very unstable
ref #6. Works fine