Specifying input by id to try and fix failing test