Added fix for lab 9.