Fixed warn issue Fixes #3