Code was looking for an ID not class

Fixes #1121