Fixed bug! (in assignment)