Finally found a way to make tests work