fixed build and issue #5