Fix extension doctrine mapping + units tests