Small modification; fix build