still trying to make tests work