Fixed minor bug from the workshop project.