Fixed bug

Fix #126