Fix issue #2

not it's clean