Fix #5

Fix #5