Fix dependency

Fix #2.