Fixed bug

fixed #18