That sounds better. I was getting a method missing so I added this check.