fixed nil: nil class error