Use a more robust way to define the dependency.
Suggested by mat, thanks for this.