Fix method missing lookup of formula class

Based on the readme example, I think this should be this