Added missing dependency

This caused a fatal error after registering