Fix #2

Fix #2