Fix in waitUntilButtonIsActive to throw expected TimeOutException