Working on actually unit testing this…