Fix for lab 3