Fix build from maven

Added missing imports