Unit Test Error in exercise 2 of mprovise_a_Jazz_Solo_with_an_LSTM_Network_v4

Im getting this error in exercise 2 when running unit test.
Test failed at index 1
Expected value

[‘TensorFlowOpLayer’, [(None, 90)], 0]

does not match the input value:

[‘Lambda’, (None, 90), 0]

AssertionError Traceback (most recent call last)
in
1 # UNIT TEST
2 output = summary(model)
----> 3 comparator(output, djmodel_out)

~/work/W1A3/test_utils.py in comparator(learner, instructor)
24 “\n\n does not match the input value: \n\n”,
25 colored(f"{b}", “red”))
—> 26 raise AssertionError(“Error in test”)
27 print(colored(“All tests passed!”, “green”))
28

AssertionError: Error in test

Do not use a Lambda layer.

Thanks for the help.