Trying to make tests work