Managed to make a test work