Fix issue with entrypoint