Should have fixed error with missing Layout class