Fix #1

Fix build error