fixed problem with solution