Fixed problem with Units page