Removed mockito, wasnt needed anyways