Fix master

@dknochen @MatthiasBrantner Not sure why this test is failing now.