Fix schedule tests