Add unitqueResult call