Lite mer fix med Unit Of Work